package io.renren.common.utils;

import java.text.SimpleDateFormat;
import java.util.*;


import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.ExcelImportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.ImportParams;
import cn.afterturn.easypoi.excel.entity.enmus.ExcelType;
import org.apache.commons.lang3.StringUtils;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
	import java.io.File;
	import java.io.IOException;
	import java.net.URLEncoder;

public class ExcelUtil{
	    public static void exportExcel(List<?> list, String title, String sheetName, Class<?> pojoClass,
	                                   String fileName, boolean isCreateHeader, HttpServletResponse response){
	        ExportParams exportParams = new ExportParams(title, sheetName);
	        exportParams.setCreateHeadRows(isCreateHeader);
	        defaultExport(list, pojoClass, fileName, response, exportParams);
	    }
	 
	    public static void exportExcel(List<?> list, String title, String sheetName, Class<?> pojoClass,String fileName,
	                                   HttpServletResponse response){
	        defaultExport(list, pojoClass, fileName, response, new ExportParams(title, sheetName));
	    }
	 
	    public static void exportExcel(List<Map<String, Object>> list, String fileName, HttpServletResponse response){
	        defaultExport(list, fileName, response);
	    }
	 
	    private static void defaultExport(List<?> list, Class<?> pojoClass, String fileName,
	                                      HttpServletResponse response, ExportParams exportParams) {
	        Workbook workbook = ExcelExportUtil.exportExcel(exportParams,pojoClass,list);
	        if (workbook != null); downLoadExcel(fileName, response, workbook);
	    }
	 
	    public static void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) {
	        try {
	            response.setCharacterEncoding("UTF-8");
	            response.setHeader("content-Type", "application/vnd.ms-excel");
	            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
	            workbook.write(response.getOutputStream());
	        } catch (IOException e) {
	            //throw new NormalException(e.getMessage());
	        }
	    }
	 
	    private static void defaultExport(List<Map<String, Object>> list, String fileName, HttpServletResponse response) {
	        Workbook workbook = ExcelExportUtil.exportExcel(list, ExcelType.HSSF);
	        if (workbook != null);
	        downLoadExcel(fileName, response, workbook);
	    }
	 
	    public static <T> List<T> importExcel(String filePath,Integer titleRows,Integer headerRows, Class<T> pojoClass){
	        if (StringUtils.isBlank(filePath)){
	            return null;
	        }
	        ImportParams params = new ImportParams();
	        params.setTitleRows(titleRows);
	        params.setHeadRows(headerRows);
	        List<T> list = null;
	        try {
	            list = ExcelImportUtil.importExcel(new File(filePath), pojoClass, params);
	        }catch (NoSuchElementException e){
	            //throw new NormalException("模板不能为空");
	        } catch (Exception e) {
	            e.printStackTrace();
	            //throw new NormalException(e.getMessage());
	        } return list;
	    }
	 
		public static <T> List<T> importExcel(MultipartFile file, Integer titleRows, Integer headerRows, Class<T> pojoClass){
	        if (file == null){ return null;
	        }
	        ImportParams params = new ImportParams();
	        params.setTitleRows(titleRows);
	        params.setHeadRows(headerRows);
	        List<T> list = null;
	        try {
	            list = ExcelImportUtil.importExcel(file.getInputStream(), pojoClass, params);
	        }catch (NoSuchElementException e){
	           // throw new NormalException("excel文件不能为空");
	        } catch (Exception e) {
	            //throw new NormalException(e.getMessage());
	            System.out.println(e.getMessage());
	        }
	        return list;
	    }

		/**
		 * export导出请求头设置
		 *
		 * @param response
		 * @param workbook
		 * @param fileName
		 * @throws Exception
		 */
		public static void export(HttpServletResponse response, Workbook workbook, String fileName) throws Exception {
			response.reset();
			response.setContentType("application/x-msdownload");
			fileName = fileName + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date());
			response.setHeader("Content-disposition", "attachment; filename=" + new String(fileName.getBytes("gb2312"), "ISO-8859-1") + ".xls");
			ServletOutputStream outStream = null;
			try {
				outStream = response.getOutputStream();
				workbook.write(outStream);
			} finally {
				outStream.close();
			}
		}


	/**
	 * 一个excel 创建多个sheet
	 *
	 * @param list
	 *            多个Map key title 对应表格Title key entity 对应表格对应实体 key data
	 *            Collection 数据
	 * @return
	 */
	/*public static Workbook exportExcel(List<Map<String, Object>> list, ExcelType type) {
		Workbook workbook = getWorkbook(type,0);
		for (Map<String, Object> map : list) {
			ExcelExportService service = new ExcelExportService();
			// service.createSheetForMap(workbook, (ExportParams)map.get("title"), (List<ExcelExportEntity>)map.get("entity"), (Collection<?>)map.get("data"));
			service.createSheet(workbook, (ExportParams)map.get("title"),(Class)map.get("entity"),(Collection<?>)map.get("data"));
		}
		return workbook;
	}*/

	private static Workbook getWorkbook(ExcelType type, int size) {
		if (ExcelType.HSSF.equals(type)) {
			return new HSSFWorkbook();
		} else if (size < 100000) {
			return new XSSFWorkbook();
		} else {
			return new SXSSFWorkbook();
		}
	}
}
